package com.sxj.shop.util;

import com.sxj.shop.exception.DataAccessException;
import org.apache.commons.dbcp.BasicDataSourceFactory;

import javax.sql.DataSource;
import java.io.IOException;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.Properties;

/**
 * @author 石小俊
 * @date 2023年12月2023/12/5日 19:22
 * 根据配置文件读取数据库的配置
 * 配置文件位置:src根目录中的dataSource.properties
 */
public class JDBCUtil {

    private static DataSource ds; // 数据库连接池
    private static ThreadLocal<Connection> local; // 线程池

    static {
        local = new ThreadLocal<>();
        Properties p = new Properties();
        try {
            p.load(JDBCUtil.class.getClassLoader().getResourceAsStream("dataSource.properties"));
            ds = BasicDataSourceFactory.createDataSource(p);
        } catch (IOException e) {
            e.printStackTrace();
            throw new ExceptionInInitializerError("JDBCUtil初始化失败");
        } catch (Exception e) {
            e.printStackTrace();
            throw new ExceptionInInitializerError("JDBCUtil初始化失败");
        }
    }

    /**
     * 使用数据库连接池获取一个线程单例的Connection对象
     * @return
     */
    public static Connection getConnection() {
        // 从线程池中获取Connection对象
        Connection conn = local.get();
        try {
            // 如果没有获取到连接,说明连接尚未获取过
            if (conn == null) {
                // 则从数据库连接池中取出一个连接
                conn = ds.getConnection();
                // 当从连接池中获取到连接之后,将获取到的连接存放到线程池中
                // 当下次从线程池中获取的时候,则获取到当前连接
                local.set(conn);
            }
        } catch (SQLException e) {
            e.printStackTrace();
            throw new DataAccessException("获取连接失败");
        }
        return conn;
    }

    /**
     * 释放指定的资源
     * 该方法主要用于持久化操作之后进行释放
     * 注意点:由于在业务层需要使用事务,操作事务必须通过Connection来进行设置
     * 因此,在dao层释放资源的时候,不能对Connection进行释放
     *
     * @param ps
     * @param rs
     */
    public static void close(PreparedStatement ps, ResultSet rs) {
        if (ps != null) {
            try {
                ps.close();
            } catch (SQLException e) {
                e.printStackTrace();
                throw new DataAccessException("释放状态集失败");
            }
        }
        if (rs != null) {
            try {
                rs.close();
            } catch (SQLException e) {
                e.printStackTrace();
                throw new DataAccessException("释放结果集失败");
            }
        }
    }

    /**
     * 释放当前线程池中的连接
     * 用于事务结束之后进行操作
     */
    public static void close() {
        // 从线程池中获取当前的连接
        Connection conn = local.get();
        try {
            // 如果获取到连接,则说明尚未释放
            if (conn != null) {
                // 则需要进行释放连接
                conn.close();
                // 既然连接已经释放,则此时线程池中的连接没有意义了
                // 则从线程池中移除已经释放的连接
                local.remove();
            }
        } catch (SQLException e) {
            e.printStackTrace();
            throw new DataAccessException("释放连接失败");
        }
        // 如果没有获取到连接,说明线程池中的连接已经释放了,则无需再次释放
    }
}
